import torch.nn as nn


class FakeEmbedding(nn.Module):
    """

    Args:
        dim : dim
    Input:
        embedding : *
    Output:
        embedding : *
    """

    def __init__(self, dim):
        super(FakeEmbedding, self).__init__()
        self.embedding_dim = dim

    def forward(self, inputs):
        return inputs
